JumpReLU SAE on toy model
Play/Pause